import re

from ai import zhipuAI

# 列表正则
pattern_1 = r'\b\d+\.\s[^\n]+'


# 列表形式的结果。从AI 回答中获取结果，放到最终结果里面
def getListResult(result):
    print(result)
    if not result:
        return result
    # 使用 zhipuAI 获取结果
    aioutput = zhipuAI.getAIResult(result)
    aiList = aiFormat(pattern_1, aioutput)
    # 将list 转换为 字符串输出
    aiStr = '\n'.join(aiList)
    print(aiStr)
    return aiStr


def getAIResult(aiQuestion):
    print(aiQuestion)
    # 将result 按照
    aiResult = zhipuAI.getAIResult(aiQuestion)
    return aiResult


# 使用regex 获取 数字. 比如1. 2. 3. 4. 开头的内容
def aiFormat(pattern, aiJson):
    # 使用re.findall查找所有匹配项
    matches = re.findall(pattern, aiJson)
    return matches
